package com.vvv.designpattern.items.composite;

import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

/**
 * @author vvv
 * @date 2023-11-18 15 28
 * discription 组合模式的树枝构建角色，一个ProductComposite相当于一个节点
 * ProductComposite同时也是访问者模式的被访问具体角色
 */
@Data
@Builder
@NoArgsConstructor
@AllArgsConstructor
public class ProductComposite extends AbstractProductItem {
    // 因为我们的目的是将数据库中的商品类目信息封装成一个节点所以ProductComposite有id, pid, name属性
    private Integer id;
    private Integer pid;
    private String name;
    // 表示当前节点的子节点
    private List<AbstractProductItem> child = new ArrayList<>();

    @Override
    public void addProductItem(AbstractProductItem item) {
        this.child.add(item);
    }

    @Override
    public void delProductChild(AbstractProductItem item) {
        ProductComposite removeItem = (ProductComposite) item;
        Iterator<AbstractProductItem> iterator = child.iterator();
        while (iterator.hasNext()) {
            ProductComposite composite = (ProductComposite) iterator.next();
            if (composite.getId().equals(removeItem.getId())) {
                iterator.remove();
                break;
            }
        }
    }
}
